/* 实现异步日志缓冲区 */

#ifndef __M_BUF_H__
#define __M_BUF_H__

#include "util.hpp"
#include <vector>
#include <cassert>

namespace log
{
    #define DEFAULT_BUFFER_SIZE (1024*1024*1) //1M大小
    //在空间达到阈值前，翻倍增长，达到阈值后，线性增长
    #define THRESHOLD_BUFFER_SIZE (1024*1024*8) //阈值大小
    #define INCREMENT_BUFFER_SIZE (1024*1024*10) //增量大小

    class Buffer{
        public:
            Buffer():_buffer(DEFAULT_BUFFER_SIZE), _reader_idx(0), _writer_idx(0){}
            //向缓冲区中写入数据
            void push(const char* data, size_t len){
                //考虑缓冲区剩余空间不够的情况
                ensureEnoughSize(len);
                //1.将数据拷贝进缓冲区
                std::copy(data, data+len, &_buffer[_writer_idx]);
                //2.将当前写入位置向后偏移
                moveWriter(len);
            }
            //返回可读数据的起始位置
            const char* begin(){
                return &_buffer[_reader_idx];
            }
            //返回可写的数据长度
            size_t writerAbleSize(){
                //对于扩容思路则不需要这个接口
                //因此此接口仅仅针对提供固定大小缓冲区的接口
                return (_buffer.size() - _writer_idx);
            }
            //返回可读数据的长度
            size_t readAblesSize(){
                return (_writer_idx - _reader_idx);
            }
            //对读写指针进行向后偏移操作
            void moveWriter(size_t len){
                assert(len <= _buffer.size());
                _writer_idx += len;
            }
            void moverReader(size_t len){
                assert(len <= readAblesSize());
                _reader_idx += len;
            }
            //重置读写位置，初始化缓冲区
            void reset(){
                _reader_idx = _writer_idx = 0;
            }
            //对buffer实现交换操作
            void swap(Buffer& buffer){
                _buffer.swap(buffer._buffer);
                std::swap(buffer._reader_idx, _reader_idx);
                std::swap(buffer._writer_idx, _writer_idx);
            }
            //判断缓冲区是否为空
            bool empty(){
                return (_reader_idx == _writer_idx);
            }
        private:
            //对空间进行扩容
            void ensureEnoughSize(size_t len){
                //空间足够用，不需要扩容
                if(len < writerAbleSize()) 
                    return;
                size_t new_size = 0;
                //还未到阈值空间，成倍增长
                if(_buffer.size() < THRESHOLD_BUFFER_SIZE){
                    new_size = _buffer.size() * 2 + len;
                }
                else{ //达到阈值空间，线性增长
                    new_size = INCREMENT_BUFFER_SIZE + len;
                }
                _buffer.resize(new_size);
            }
        private:
            std::vector<char> _buffer;
            size_t _reader_idx; //当前可读数据指针 -- 实际是下标
            size_t _writer_idx; //当前可写数据指针 
    };
} // namespace name


#endif